package com.hc.programming.tree;

import java.util.Arrays;

/**
 * 给你二叉树的根节点 root 和一个表示目标和的整数 targetSum 。判断该树中是否存在 根节点到叶子节点 的路径，这条路径上所有节点值相加等于目标和 targetSum 。如果存在，返回 true ；否则，返回 false 。
 * 叶子节点 是指没有子节点的节点。
 * <p>
 * 示例 1：
 * <a href="./路径总和-示例1.png">示例1</a>
 * 输入：root = [5,4,8,11,null,13,4,7,2,null,null,null,1], targetSum = 22
 * 输出：true
 * 解释：等于目标和的根节点到叶节点路径如上图所示。
 * 示例 2：
 * <a href="./找树左下角的值-示例1.png">示例2</a>
 * 输入：root = [1,2,3], targetSum = 5
 * 输出：false
 * 解释：树中存在两条根节点到叶子节点的路径：
 * (1 --> 2): 和为 3
 * (1 --> 3): 和为 4
 * 不存在 sum = 5 的根节点到叶子节点的路径。
 * 示例 3：
 * 输入：root = [], targetSum = 0
 * 输出：false
 * 解释：由于树是空的，所以不存在根节点到叶子节点的路径。
 * <p>
 * 提示：
 * 树中节点的数目在范围 [0, 5000] 内
 * -1000 <= Node.val <= 1000
 * -1000 <= targetSum <= 1000
 *
 * @author huangchao E-mail:fengquan8866@163.com
 * @version 创建时间：2024/10/15 10:54
 */
public class 路径总和 {
    public static void main(String[] args) {
        Integer[] arr = new Integer[]{5, 4, 8, 11, null, 13, 4, 7, 2, null, null, null, 1};
        TreeNode tree = TreeNode.tree(arr);
        System.out.println(Arrays.toString(arr) + "=true,--" + hasPathSum(tree, 22));
        arr = new Integer[]{1, 2, 3};
        tree = TreeNode.tree(arr);
        System.out.println(Arrays.toString(arr) + "=false,--" + hasPathSum(tree, 5));
        arr = new Integer[]{};
        tree = TreeNode.tree(arr);
        System.out.println(Arrays.toString(arr) + "=false,--" + hasPathSum(tree, 0));
    }

    public static boolean hasPathSum(TreeNode root, int targetSum) {
        if (root == null) return false;
//        return 递归(root, targetSum, 0);
        return 递归2(root, targetSum);
    }

    private static boolean 递归2(TreeNode root, int targetSum) {
        if (root.left == null && root.right == null) {
            return root.val == targetSum;
        }
        if (root.left != null && 递归2(root.left, targetSum - root.val)) return true;
        return root.right != null && 递归2(root.right, targetSum - root.val);
    }

    private static boolean 递归(TreeNode root, int targetSum, int sum) {
        sum += root.val;
        if (root.left == null && root.right == null) {
            return sum == targetSum;
        }
        if (root.left != null && 递归(root.left, targetSum, sum)) {
            return true;
        }
        return root.right != null && 递归(root.right, targetSum, sum);
    }
}
